import torchvision
from torchsummary import summary
vgg16 = torchvision.models.vgg16()

summary(vgg16,input_size=(3,64,64),device="cpu")
